#include "UdpServer.hpp"
#include "User.hpp"

// ./server_udp localip localport
int main(int argc, char *argv[])
{
    if(argc != 2)
    {
        std::cerr << "Usage: " << argv[0] << " localPort" << std::endl;
        Die(USAGE_ERR);
    }
    uint16_t port = std::stoi(argv[1]);

    ENABLE_CONSOLE_LOG();

    // 用户管理模块
    std::shared_ptr<UserManage> um = std::make_shared<UserManage>();
    // 网络模块
    std::unique_ptr<UdpServer> svr_uptr = std::make_unique<UdpServer>([&um](InetAddr &id){
                                                                    um->AddUser(id);}, port);
    svr_uptr->RegisterService([&um](InetAddr &id) {um->AddUser(id);},
                            [&um](int sockfd, const std::string &message){um->Router(sockfd, message);},
                            [&um](InetAddr &id) {um->DelUser(id);});
    
    svr_uptr->InitServer();
    
    svr_uptr->Start();
    return 0;
}